def retry(count):
    def wrapper(func):
        def decorate(*args, **kwargs):
            for i in range(count):
                try:
                    return func(*args, **kwargs)
                except Exception as exp:
                    raise exp

        return decorate

    return wrapper
